#pragma once

#include <opencv2\opencv.hpp>

using namespace cv;

class EMax
{
private:
	EM em;

public:

	void set_trainer(EM _em)
	{
		em = _em;
	}

	EM get_trainer()
	{
		return em;
	}

	void train(const Mat& data, const Mat& label, const EM& params)
	{
		em.train(data, noArray(), label, noArray());
	}

	float predict(const Mat& sample)
	{
		return em.predict(sample)[1];
	}
};